/*
 * Copyright (C) 2018-2021. Huawei Technologies Co., Ltd. All rights reserved.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.prestosql.plugin.hive;

public enum HiveACIDWriteType
{
    VACUUM(-2),
    VACUUM_UNIFY(-2),
    NONE(-1),
    INSERT(0),
    INSERT_OVERWRITE(0),
    UPDATE(0), //Hive ACID semantics post hive-3.x expects operation to be 0.
    DELETE(2);

    private int operationId;

    HiveACIDWriteType(int operationId)
    {
        this.operationId = operationId;
    }

    public int getOperationId()
    {
        return operationId;
    }

    public static boolean isUpdateOrDelete(HiveACIDWriteType writeType)
    {
        return writeType == UPDATE || writeType == DELETE;
    }

    public static boolean isRowIdNeeded(HiveACIDWriteType writeType)
    {
        return isUpdateOrDelete(writeType) || isVacuum(writeType);
    }

    public static boolean isVacuum(HiveACIDWriteType writeType)
    {
        return writeType == VACUUM || writeType == VACUUM_UNIFY;
    }
}
